data_path = 'data/cardio_sampled.csv'
preprocessed_data_path = 'data/cardio_preprocessed.csv'

figure_path = {
    'feature_distributions': 'data_figure/figure1.png',
    'box_plot': 'data_figure/figure2.png',
    'scatter_plot': 'data_figure/figure3.png',
    'HeatMap': 'data_figure/HeatMap.png'
}

outlier_removal_method = 'IQR'
numerical_features = ['height', 'weight', 'ap_hi', 'ap_lo']

categorical_features = ['gender', 'cholesterol', 'gluc']
binary_features = ['smoke', 'alco', 'active']

scale_numerical_features = True
scaling_method = 'StandardScaler'

test_size = 0.2
random_state = 8

target_variable = 'cardio'

train_test_data_paths = {
    'X_train': 'data/X_train.csv',
    'y_train': 'data/y_train.csv',
    'X_test': 'data/X_test.csv',
    'y_test': 'data/y_test.csv',
}

data_paths = {
    'X_train': 'data/cardio_xtrain_preprocessed.csv',
    'y_train': 'data/y_train.csv',
    'X_test': 'data/cardio_xtest_preprocessed.csv',
    'y_test': 'data/y_test.csv',
}

config = {
    'input_size': 10,
    'dropout_rate': 0.5,
    'learning_rate': 0.0001,
    'batch_size': 64,
    'epochs': 200,
    'random_state': 42
}

result_path = {
    'Loss Over Epochs': 'results/Loss.png',
    'Accuracy Over Epochs': 'results/Accuracy.png',
    'Metrics Over Epochs': 'results/Metrics_Over_Epochs.png',
    'ROC': 'results/ROC.png',
    'Confusion Matrix': 'results/Confusion Matrix.png'
}
